#include <gtest/gtest.h>
#include <plog/Log.h>
#include "trt_define.h"

#include "trt_builder/trt_builder.h"
#include "utils/logger.h"

_TRT_INFER_BEGIN

class BuilderTest : public ::testing::Test
{
protected:
    BuilderTest() {}
    ~BuilderTest() override {}
    void SetUp() override {}
    void TearDown() override {}
};

TEST_F(BuilderTest, test_builder)
{
    LOGER_INST.init();
    LOG_INFO << "test_builder";
    std::string source_model_path = "D:\\learnSpace\\trt_infer\\test_data\\classifier.onnx";
    std::string save_model_path = "D:\\learnSpace\\trt_infer\\test_data\\classifier.engine";
    TrtBuilder::compile(TrtBuilder::ModelType::FP32, 1, source_model_path, save_model_path, 1ul << 30);
}


_TRT_INFER_END